import torch
import numpy
x = torch.arange(12)
print(x.shape)

print(x.numel())